package com.yebxxx.server.controller;


import cn.afterturn.easypoi.excel.ExcelExportUtil;
import cn.afterturn.easypoi.excel.ExcelImportUtil;
import cn.afterturn.easypoi.excel.entity.ExportParams;
import cn.afterturn.easypoi.excel.entity.ImportParams;
import cn.afterturn.easypoi.excel.entity.enmus.ExcelType;
import com.yebxxx.config.DateConventer;
import com.yebxxx.server.pojo.*;
import com.yebxxx.server.service.*;
import io.swagger.annotations.ApiOperation;
import org.apache.poi.ss.usermodel.Workbook;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.web.bind.annotation.*;
import org.springframework.web.multipart.MultipartFile;

import javax.servlet.ServletOutputStream;
import javax.servlet.http.HttpServletResponse;
import java.io.IOException;
import java.net.URLEncoder;
import java.text.DateFormat;
import java.time.LocalDate;
import java.util.List;

/**
 * <p>
 *  前端控制器
 * </p>
 *
 * @author stx
 * @since 2022-03-23
 */
@RestController
@RequestMapping("/employee/basic")
public class EmployeeController {
    @Autowired
    private IEmployeeService employeeService;
    @Autowired
    private IPoliticsStatusService politicsStatusService;
    @Autowired
    private IJoblevelService joblevelService;
    @Autowired
    private INationService nationService;
    @Autowired
    private IPositionService positionService;
    @Autowired
    private  IDepartmentService departmentService;
    @ApiOperation(value = "获取员工信息(分页)")
    @GetMapping("/")
    public PageBean getEmployee(@RequestParam(defaultValue = "1" )Integer currentPage,
                                @RequestParam(defaultValue = "10")Integer size,
                                Employee employee, LocalDate[] beginDateScope){
        return employeeService.getEmployeeByPage(currentPage,size,employee,beginDateScope);

    }
    @ApiOperation("获取所有政治面貌")
    @GetMapping("/politicsStatus")
    public List<PoliticsStatus> getAllPoliticsStatus(){
        return politicsStatusService.list();
    }

    @ApiOperation("获取所有职称")
    @GetMapping("/jobLevels")
    public  List<Joblevel> getAllJobLevel(){
        return  joblevelService.list();
    }

    @ApiOperation("获取所有民族")
    @GetMapping("/nations")
    public  List<Nation> getAllNations(){
        return  nationService.list();
    }

    @ApiOperation("获取所有职位")
    @GetMapping("/positions")
    public  List<Position> getAllPositions(){
        return  positionService.list();
    }

    @ApiOperation("获取所有部门")
    @GetMapping("/deps")
    public  List<Department> getAllDeps(){
        return  departmentService.getAllDepartment();
    }

    @ApiOperation("获取工号")
    @GetMapping("/MaxWorkId")
    public RespBean maxWorkId(){
        return employeeService.maxWorkId();
    }

    @ApiOperation("添加员工")
    @PostMapping("/")
    public  RespBean addEmp(@RequestBody Employee employee){
        return  employeeService.addEmp(employee);
    }
    @ApiOperation(value = "更新员工信息")
    @PutMapping("/")
    public  RespBean updateEmp(@RequestBody Employee employee){
        if(employeeService.updateById(employee)){
            return  RespBean.success("更新成功！");
        }
        return  RespBean.error("更新失败！");
    }
    @ApiOperation(value = "删除员工信息")
    @DeleteMapping("/{id}")
    public RespBean deleteEmp(@PathVariable Integer id){
        if(employeeService.removeById(id)){
            return RespBean.success("删除成功！");
        }
        return  RespBean.error("删除失败！");
    }
    @ApiOperation(value = "导出员工数据")
    @GetMapping(value = "/export",produces = "application/octet-stream")
    public void exportEmployee(HttpServletResponse response){
        List<Employee> employeeList = employeeService.getEmployee(null);
        ExportParams exportParams=new ExportParams("员工表","员工表", ExcelType.HSSF);
        Workbook workbook = ExcelExportUtil.exportExcel(exportParams, Employee.class, employeeList);
        ServletOutputStream outputStream=null;
        try {
            response.setHeader("content-type","application/octet-stream");
            response.setHeader("content-disposition","attachment;filename="+ URLEncoder.encode("员工表.xls","UTF-8"));
             outputStream = response.getOutputStream();
             workbook.write(outputStream);
        } catch (IOException e) {
            e.printStackTrace();
        }finally {
            if(null!=outputStream){
                try {
                    outputStream.close();
                } catch (IOException e) {
                    e.printStackTrace();
                }
            }
        }

    }

    @ApiOperation(value = "导入员工数据")
    @PostMapping("/import")
    public  RespBean importEmp(MultipartFile file){
        ImportParams params=new ImportParams();
        params.setTitleRows(1); //去掉标题行
        List<Nation> nationList = nationService.list();
        List<Position> positionList = positionService.list();
        List<Department> departmentList = departmentService.list();
        List<Joblevel> joblevelList = joblevelService.list();
        List<PoliticsStatus> politicsStatusList = politicsStatusService.list();
        try {
            List<Employee> list = ExcelImportUtil.importExcel(file.getInputStream(), Employee.class, params);
            list.forEach(employee->{
                //民族id
                Integer id = nationList.get(nationList.indexOf(new Nation(employee.getNation().getName()))).getId();
                employee.setNationId(id);

                //政治面貌id
              employee.setPoliticId( politicsStatusList.get(  politicsStatusList.indexOf(new PoliticsStatus(employee.getPoliticsStatus().getName()))).getId());

              //部门id
                employee.setDepartmentId(departmentList.get(departmentList.indexOf(new Department(employee.getDepartment().getName()))).getId());
                //职称id
            employee.setJobLevelId(joblevelList.get(joblevelList.indexOf(new Joblevel(employee.getJoblevel().getName()))).getId());
            //职位id
             employee.setPosId(positionList.get(positionList.indexOf(new Position(employee.getPosition().getName()))).getId());

            });
            if(employeeService.saveBatch(list)){
                return RespBean.success("导入成功！");
            }

        } catch (Exception e) {
            e.printStackTrace();
        }
        return  RespBean.error("导入失败！");
    }

    @ApiOperation(value = "员工离职")
    @PostMapping("/quit")
    public RespBean deleteEm(Integer id, String notworkwhy){
        Employee employee=new Employee();
        employee.setId(id);
        employee.setNotWorkDate(LocalDate.now());
        employee.setNotworkwhy(notworkwhy);
        return employeeService.updateWorkState(employee);
    }
//    public PageBean getEmployee(@RequestParam(defaultValue = "1" )Integer currentPage,
//                                @RequestParam(defaultValue = "10")Integer size,
//                                Employee employee, LocalDate[] beginDateScope){
//        return employeeService.getEmployeeByPage(currentPage,size,employee,beginDateScope);
//
//    }
    @ApiOperation(value = "离职员工列表")
    @PostMapping("/quit/EmpList")
    public  PageBean deleteEmList(@RequestParam(defaultValue = "1" )Integer currentPage,
                                  @RequestParam(defaultValue = "10")Integer size,

                                  EmployeeTrain employeeTrain,
                                  Teacher teacher){
        return employeeService.deleteEmList(currentPage,size,employeeTrain,teacher);
    }
}
